var _ = require('lodash');

function stateFilter() {
    return function(text) {
        if(_.isEqual(text, 0)){
            return "Untreated";
        }else if(_.isEqual(text, 1)){
            return "Pass";
        }else if(_.isEqual(text, 2)){
            return "Refuse";
        }
    }
}

module.exports = stateFilter;